"""
@author: hjf
@file: read-image.py
@time: 2021/11/8 10:33
@desc: 常用的读图像的方法
"""

import os
import cv2
import matplotlib.pyplot as plt

if __name__ == '__main__':
    url = 'C:/Users/hejianfei/Desktop/R-C.jpg'
    # 调用imread函数读取图像，此时读入的通道顺序默认是BGR,需要使用cv2.cvtColor函数转换为RGB顺序
    # 使用cv2.imread函数时，可以传入相应的参数
    # cv2.IMREAD_COLOR：代表读取彩色图片，默认的取值，结果是一个三维的ndarray数组
    # cv2.IMREAD_GRAYSCALE: 代表读取灰度图片，结果是一个二维的ndarray数组
    image = cv2.imread(url, cv2.IMREAD_COLOR)
    # 如果使用cv2读取彩色图片，显示彩色图片，则不需要转换通道
    cv2.imshow('img', image)
    cv2.waitKey(0)
    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    # 将图像转换为灰度图像
    img = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
    # 使用plt.imshow函数显示灰度图像时，一定要加cmap='gray',否则显示会不正确
    plt.imshow(img, cmap='gray')
    plt.show()

